numpy.where function

join shbcf.ru